import argparse
from configs.config import global_var


def load_predict_config(para):
    # 设置参数
    parser = argparse.ArgumentParser()
    parser.add_argument('--mode', type=str, default='标准正位', help='the mode of gesture')
    parser.add_argument('--CT_path', type=str, default=para + "CT_data", help='CT_slice_data location')
    parser.add_argument('--CT_pt_path', type=str, default=para + "PointClouds/spine107_pts.npy",
                        help='CT_pt_data location')
    parser.add_argument('--model_path', type=str,
                        default=para + 'data/train_history/基础模型_均匀5+5_第二次实验/model/6D_model_loss0.0010012350976467133.pth',
                        help='load trained model to predictor')
    parser.add_argument('--num_classes', type=int, default=6,
                        help='saving result path')
    parser.add_argument('--saving_drr_path', type=str, default=global_var.saving_address,
                        help='saving result path')

    args = parser.parse_args()
    para_dict = {
        'CT_path': args.CT_path,
        'CT_pt_path': args.CT_pt_path,
        'model_path': args.model_path,
        'mode': args.mode,
        'num_classes': args.num_classes,
        'saving_drr_path': args.saving_drr_path
    }
    return para_dict
